"""
Visualization Module

提供PRISM系统的各种可视化功能：
- 特征图可视化
- Grad-CAM类别激活图
- 统计图表生成
"""

from .feature_extractor import (
    FeatureExtractor,
    OverLoCKFeatureExtractor,
    DINOv2FeatureExtractor,
    FeatureVisualizer
)

from .gradcam import (
    GradCAM,
    GradCAMPlusPlus,
    MultiLayerGradCAM,
    GradCAMVisualizer
)

from .statistics import StatisticsPlotter

__all__ = [
    # Feature Extraction
    'FeatureExtractor',
    'OverLoCKFeatureExtractor',
    'DINOv2FeatureExtractor',
    'FeatureVisualizer',

    # Grad-CAM
    'GradCAM',
    'GradCAMPlusPlus',
    'MultiLayerGradCAM',
    'GradCAMVisualizer',

    # Statistics
    'StatisticsPlotter',
]
